from app import create_app
from app.models import User

def query_user_ids_by_room(room_number):
    app = create_app()
    with app.app_context():
        # 查询指定房间号的所有用户
        users_in_room = User.query.filter_by(room_number=room_number).all()
        
        if not users_in_room:
            print(f"房间号为 {room_number} 的用户不存在")
            return
        
        # 获取用户数量
        count = len(users_in_room)
        
        # 打印查询结果
        print(f"房间号为 {room_number} 的用户共有 {count} 位，信息如下：")
        # 直接遍历用户对象，同时获取每个用户的ID和签名
        for user in users_in_room:
            print(f"- ID: {user.id} -- 签名: {user.signature}")

if __name__ == "__main__":
    # 在这里指定要查询的房间号
    target_room_number = "1号楼 3单元 602号门"  # 替换为实际要查询的房间号
    query_user_ids_by_room(target_room_number)
    